/*
 * BEEM is a videoconference application on the Android Platform.
 * Copyright (C) 2009 by Frederic-Charles Barthelery,
 * Jean-Manuel Da Silva,
 * Nikita Kozlov,
 * Philippe Lago,
 * Jean Baptiste Vergely,
 * Vincent Veronis.
 * This file is part of BEEM.
 * BEEM is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 * BEEM is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 * You should have received a copy of the GNU General Public License
 * along with BEEM. If not, see <http://www.gnu.org/licenses/>.
 * Please send bug reports with examples or suggestions to
 * contact@beem-project.com or http://dev.beem-project.com/
 */
package com.beem.project.beem.smack.avatar;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;

import org.jivesoftware.smack.util.Cache;

/**
 * An avatar cache which store the avatars in memory.
 */
public class MemoryAvatarCache implements AvatarCache {

	private static final int BUFFER_SIZE = 1024;
	private Cache<String, byte[]> mCache;

	/**
	 * Create a MemoryAvatarCache.
	 * 
	 * @param maxSize
	 *            the maximum number of objects the cache will hold. -1 means the cache has no max size.
	 * @param maxlifetime
	 *            the maximum amount of time (in ms) objects can exist in cache before being deleted.
	 *            -1 means objects never expire.
	 */
	public MemoryAvatarCache(final int maxSize, final long maxlifetime) {
		mCache = new Cache<String, byte[]>(maxSize, maxlifetime);
	}

	@Override
	public void put(String key, byte[] data) throws IOException {
		mCache.put(key, data);
	}

	@Override
	public void put(String key, InputStream in) throws IOException {
		ByteArrayOutputStream os = new ByteArrayOutputStream();
		try {
			byte[] data = new byte[BUFFER_SIZE];
			int nbread;
			while ((nbread = in.read(data)) != -1)
				os.write(data, 0, nbread);
		} finally {
			in.close();
			os.close();
		}
		mCache.put(key, os.toByteArray());
	}

	@Override
	public byte[] get(String key) throws IOException {
		return mCache.get(key);
	}

	@Override
	public boolean contains(String key) {
		return mCache.containsKey(key);
	}
}
